import random
import unittest

from api.ihrm_emp import IHRMEmpApi
from api.ihrm_login_api import IHRMLoginApi
from utils.db_utils import DBUtils


class TestIHRMEmp(unittest.TestCase):
    def test_emp(self):
        # 登录
        resp = IHRMLoginApi.login_request({
            'mobile': '13800000002',
            'password': '123456'
        })
        token = resp.json().get('data')
        # 添加
        ran_mobile = random.randint(10000000, 99999999)
        resp = IHRMEmpApi.add_emp(token, {
            'username': '大黄',
            'mobile': f'138{ran_mobile}',
            'workNumber': '7788'
        })
        emp_id = resp.json().get('data').get('id')
        # 查询
        resp = IHRMEmpApi.query_emp(token, emp_id)
        # 对resp进行断言
        # 修改
        resp = IHRMEmpApi.update_emp(token, emp_id, {
            'username': '小黄'
        })
        # 对resp进行断言
        # 查询
        resp = IHRMEmpApi.query_emp(token, emp_id)
        # 对resp进行断言
        # 删除
        resp = IHRMEmpApi.del_emp(token, emp_id)
        # 对resp进行断言
        # 数据库查询
        results = DBUtils.query_db(f'select * from 员工表 where id={emp_id}')
        # 对results进行断言
